"""
RINN: Rank Inspired Neural Network for solving linear partial differential equations.
Based on Paper 2506.17654.
"""

__version__ = "1.0.0"
__author__ = "RINN Implementation Team"

# Import main components
from .pielm.solver import PIELMNetwork, PIELMSolver, create_pielm_solver
from .rinn.pretrainer import RINNPretrainer  
from .rinn.rinn_es import RINNEarlyStopping, create_rinn_es_solver
from .common import utils, metrics, plotting

__all__ = [
    'PIELMNetwork',
    'PIELMSolver', 
    'create_pielm_solver',
    'RINNPretrainer',
    'RINNEarlyStopping',
    'create_rinn_es_solver',
    'utils',
    'metrics',
    'plotting'
]